

import numpy as np


#  5 classes
label_weights_list = [0 ,1 ,1 , 1 ,0]

labels_batch = np.array([
    [1],[4],[0],[3],[1],[0],[2],[4],[0],[3],[2]
])


weights_batch = np.array(label_weights_list)[labels_batch]
# weights_batch = np.array(label_weights_list)[0]
print(weights_batch)